TS 强类型非常好用,但在实际运用中,免不了遇到一些难以描述,反复看官方文档也解决不了的问题,至今为止也没有任何一篇文档,或者一套教材可以解决所有犄角旮旯的类型问题。为什么会这样呢?因为 TS 并不是简单的注释器,而是一门图灵完备的语言,所以很多问题的解决方法藏在基础能力里,但你学会了基础能力又不一定能想到这么用。
解决该问题的最好办法就是多练,通过实际案例不断刺激你的大脑,让你养成 TS 思维习惯。所以话不多说,我们今天从 type-challenges 的 Easy 难度题目开始吧。
手动实现内置 Pick<T, K>
函数,返回一个新的类型,从对象 T 中抽取类型 K:
interface Todo {
title: string
description: string
completed: boolean
}
type TodoPreview = MyPick<Todo, 'title' | 'completed'>
const todo: TodoPreview = {
title: 'Clean room',
completed: false,
}
结合例子更容易看明白,也就是 K
是一个字符串,我们需要返回一个新类型,仅保留 K
定义的 Key。
第一个难点在如何限制 K
的取值,比如传入 T
中不存在的值就要报错。这个考察的是硬知识,只要你知道 A extends keyof B
这个语法就能联想到。
第二个难点在于如何生成一个仅包含 K
定义 Key 的类型,你首先要知道有 { [A in keyof B]: B[A] }
这个硬知识,这样可以重新组合一个对象:
// 代码 1
type Foo<T> = {
[P in keyof T]: T[P]
}
只懂这个语法不一定能想出思路,原因是你要打破对 TS 的刻板理解,[K in keyof T]
不是一个固定模板,其中 keyof T
只是一个指代变量,它可以被换掉,如果你换掉成另一个范围的变量,那么这个对象的 Key 值范围就变了,这正好契合本题的 K
:
// 代码 2(本题答案)
type MyPick<T, K extends keyof T> = {
[P in K]: T[P]
}
这个题目别看知道答案后简单,回顾下还是有收获的。对比上面两个代码例子,你会发现,只不过是把代码 1 的 keyof T
从对象描述中提到了泛型定义里而已,所以功能上没有任何变化,但因为泛型可以由用户传入,所以代码 1 的 P in keyof T
因为没有泛型支撑,这里推导出来的就是 T
的所有 Keys,而代码 2 虽然把代码挪到了泛型,但因为用的是 extends
描述,所以表示 P
的类型被约束到了 T
的 Keys,至于具体是什么,得看用户代码怎么传。
所以其实放到泛型里的 K
是没有默认值的,而写到对象里作为推导值就有了默认值。泛型里给默认值的方式如下:
// 代码 3
type MyPick<T, K extends keyof T = keyof T> = {
[P in K]: T[P]
}
也就是说,这样 MyPick<Todo>
就也可以正确工作并原封不动返回 Todo
类型,也就是说,代码 3 在不传第二个参数时,与代码 1 的功能完全一样。仔细琢磨一下共同点与区别,为什么代码 3 可以做到和代码 1 功能一样,又有更强的拓展性,你对 TS 泛型的实战理解就上了一个台阶。
手动实现内置 Readonly<T>
函数,将对象所有属性设置为只读:
interface Todo {
title: string
description: string
}
const todo: MyReadonly<Todo> = {
title: "Hey",
description: "foobar"
}
todo.title = "Hello" // Error: cannot reassign a readonly property
todo.description = "barFoo" // Error: cannot reassign a readonly property
这道题反而比第一题简单,只要我们用 { [A in keyof B]: B[A] }
重新声明对象,并在每个 Key 前面加上 readonly
修饰即可:
// 本题答案
type MyReadonly<T> = {
readonly [K in keyof T]: T[K]
}
根据这个特性我们可以做很多延伸改造,比如将对象所有 Key 都设定为可选:
type Optional<T> = {
[K in keyof T]?: T[K]
}
{ [A in keyof B]: B[A] }
给了我们描述每一个 Key 属性细节的机会,限制我们发挥的只有想象力。
实现类型 First<T>
,取到数组第一项的类型:
type arr1 = ['a', 'b', 'c']
type arr2 = [3, 2, 1]
type head1 = First<arr1> // expected to be 'a'
type head2 = First<arr2> // expected to be 3
这题比较简单,很容易想到的答案:
// 本题答案
type First<T extends any[]> = T[0]
但在写这个答案时,有 10% 脑细胞提醒我没有判断边界情况,果然看了下答案,有空数组的情况要考虑,空数组时返回类型 never
而不是 undefined
会更好,下面几种写法都是答案:
type First<T extends any[]> = T extends [] ? never : T[0]
type First<T extends any[]> = T['length'] extends 0 ? never : T[0]
type First<T> = T extends [infer P, ...infer Rest] ? P : never
第一种写法通过 extends []
判断 T
是否为空数组,是的话返回 never
。
第二种写法通过长度为 0 判断空数组,此时需要理解两点:1. 可以通过 T['length']
让 TS 访问到值长度(类型的),2. extends 0
表示是否匹配 0,即 extends
除了匹配类型,还能直接匹配值。
第三种写法是最省心的,但也使用了 infer
关键字,即使你充分知道 infer
怎么用(精读《Typescript infer 关键字》),也很难想到它。用 infer
的理由是:该场景存在边界情况,最便于理解的写法是 “如果 T 形如 <P, ...>
” 那我就返回类型 P
,否则返回 never
”,这句话用 TS 描述就是:T extends [infer P, ...infer Rest] ? P : never
。
实现类型 Length<T>
获取元组长度:
type tesla = ['tesla', 'model 3', 'model X', 'model Y']
type spaceX = ['FALCON 9', 'FALCON HEAVY', 'DRAGON', 'STARSHIP', 'HUMAN SPACEFLIGHT']
type teslaLength = Length<tesla> // expected 4
type spaceXLength = Length<spaceX> // expected 5
经过上一题的学习,很容易想到这个答案:
type Length<T extends any[]> = T['length']
对 TS 来说,元组和数组都是数组,但元组对 TS 来说可以观测其长度,T['length']
对元组来说返回的是具体值,而对数组来说返回的是 number
。
实现类型 Exclude<T, U>
,返回 T
中不存在于 U
的部分。该功能主要用在联合类型场景,所以我们直接用 extends
判断就行了:
// 本题答案
type Exclude<T, U> = T extends U ? never : T
实际运行效果:
type C = Exclude<'a' | 'b', 'a' | 'c'> // 'b'
看上去有点不那么好理解,这是因为 TS 对联合类型的执行是分配律的,即:
Exclude<'a' | 'b', 'a' | 'c'>
// 等价于
Exclude<'a', 'a' | 'c'> | Exclude<'b', 'a' | 'c'>
实现类型 Awaited
,比如从 Promise<ExampleType>
拿到 ExampleType
。
首先 TS 永远不会执行代码,所以脑子里不要有 “await 得等一下才知道结果” 的念头。该题关键就是从 Promise<T>
中抽取类型 T
,很适合用 infer
做:
type MyAwaited<T> = T extends Promise<infer U> ? U : never
然而这个答案还不够标准,标准答案考虑了嵌套 Promise
的场景:
// 该题答案
type MyAwaited<T extends Promise<unknown>> = T extends Promise<infer P>
? P extends Promise<unknown> ? MyAwaited<P> : P
: never
如果 Promise<P>
取到的 P
还形如 Promise<unknown>
,就递归调用自己 MyAwaited<P>
。这里提到了递归,也就是 TS 类型处理可以是递归的,所以才有了后面版本做尾递归优化。
实现类型 If<Condition, True, False>
,当 C
为 true
时返回 T
,否则返回 F
:
type A = If<true, 'a', 'b'> // expected to be 'a'
type B = If<false, 'a', 'b'> // expected to be 'b'
之前有提过,extends
还可以用来判定值,所以果断用 extends true
判断是否命中了 true
即可:
// 本题答案
type If<C, T, F> = C extends true ? T : F
用类型系统实现 Concat<P, Q>
,将两个数组类型连起来:
type Result = Concat<[1], [2]> // expected to be [1, 2]
由于 TS 支持数组解构语法,所以可以大胆的尝试这么写:
type Concat<P extends any[], Q extends any[]> = [...P, ...Q]
考虑到 Concat
函数应该也能接收非数组类型,所以做一个判断,为了方便书写,把 extends
从泛型定义位置挪到 TS 类型推断的运行时:
// 本题答案
type Concat<P, Q> = [
...P extends any[] ? P : [P],
...Q extends any[] ? Q : [Q],
]
解决这题需要信念,相信 TS 可以像 JS 一样写逻辑。这些能力都是版本升级时渐进式提供的,所以需要不断阅读最新 TS 特性,快速将其理解为固化知识,其实还是有一定难度的。
用类型系统实现 Includes<T, K>
函数:
type isPillarMen = Includes<['Kars', 'Esidisi', 'Wamuu', 'Santana'], 'Dio'> // expected to be `false`
由于之前的经验,很容易做下面的联想:
// 如果题目要求是这样
type isPillarMen = Includes<'Kars' | 'Esidisi' | 'Wamuu' | 'Santana', 'Dio'>
// 那我就能用 extends 轻松解决了
type Includes<T, K> = K extends T ? true : false
可惜第一个输入是数组类型,extends
可不支持判定 “数组包含” 逻辑,此时要了解一个新知识点,即 TS 判断中的 [number]
下标。不仅这道题,以后很多困难题都需要它作为基础知识。
[number]
下标表示任意一项,而 extends T[number]
就可以实现数组包含的判定,因此下面的解法是有效的:
type Includes<T extends any[], K> = K extends T[number] ? true : false
但翻答案后发现这并不是标准答案,还真找到一个反例:
type Includes<T extends any[], K> = K extends T[number] ? true : false
type isPillarMen = Includes<[boolean], false> // true
原因很简单,true
、false
都继承自 boolean
,所以 extends
判断的界限太宽了,题目要求的是精确值匹配,故上面的答案理论上是错的。
标准答案是每次判断数组第一项,并递归(讲真觉得这不是 easy 题),分别有两个难点。
第一如何写 Equal 函数?比较流行的方案是这个:
type Equal<X, Y> =
(<T>() => T extends X ? 1 : 2) extends
(<T>() => T extends Y ? 1 : 2) ? true : false
关于如何写 Equal 函数还引发了一次 小讨论,上面的代码构造了两个函数,这两个函数内的 T
属于 deferred(延迟)判断的类型,该类型判断依赖于内部 isTypeIdenticalTo
函数完成判断。
有了 Equal
后就简单了,我们用解构 + infer
+ 递归的方式做就可以了:
// 本题答案
type Includes<T extends any[], K> =
T extends [infer F, ...infer Rest] ?
Equal<F, K> extends true ?
true
: Includes<Rest, K>
: false
每次取数组第一个值判断 Equal
,如果不匹配则拿剩余项递归判断。这个函数组合了不少 TS 知识,比如:
- 递归
- 解构
infer
extends true
可以发现,就为了解决 true extends boolean
为 true
的问题,我们绕了一大圈使用了更复杂的方式来实现,这在 TS 体操中也算是常态,解决问题需要耐心。
实现 Push<T, K>
函数:
type Result = Push<[1, 2], '3'> // [1, 2, '3']
这道题真的很简单,用解构就行了:
// 本题答案
type Push<T extends any[], K> = [...T, K]
可见,想要轻松解决一个 TS 简单问题,首先你需要能解决一些困难问题 😁。
实现 Unshift<T, K>
函数:
type Result = Unshift<[1, 2], 0> // [0, 1, 2,]
在 Push
基础上改下顺序就行了:
// 本题答案
type Unshift<T extends any[], K> = [K, ...T]
实现内置函数 Parameters
:
Parameters
可以拿到函数的参数类型,直接用 infer
实现即可,也比较简单:
type Parameters<T> = T extends (...args: infer P) => any ? P : []
infer
可以很方便从任何具体的位置取值,属于典型难懂易用的语法。
学会 TS 基础语法后,活用才是关键。
如果你想参与讨论,请 点击这里,每周都有新的主题,周末或周一发布。前端精读 - 帮你筛选靠谱的内容。
关注 前端精读微信公众号
版权声明:自由转载-非商用-非衍生-保持署名(创意共享 3.0 许可证)